package frontdesk

import (
	"individual/controllers"
	"individual/models"
)

type LinkController struct {
	controllers.BaseController
}

// GetDirectLinks
// @Title	获取自定连接
// @Description 获取自定连接信息
// @Success 201 获取自定连接成功
// @Failure 403 获取自定连接失败
// @router /getDirectLinks/ [get]
func (c *LinkController) GetDirectLinks() {
	var dirLinks []models.DirectLink
	_, err := c.O.QueryTable(new(models.DirectLink).TableName()).All(&dirLinks)
	if err != nil {
		c.Data["json"] = models.GetRevaErr(err, "获取自定义连接失败")
		c.ServeJSON()
		return
	}
	c.Data["json"] = models.GetRevaOk(dirLinks)
	c.ServeJSON()
}

// GetFriendLinks
// @Title	获取友链
// @Description 获取友链
// @Success 201 获取友链成功
// @Failure 403 获取友链失败
// @router /getFriendLinks/ [get]
func (c *LinkController) GetFriendLinks() {
	var friendLinks []models.FriendLink
	_, err := c.O.QueryTable(new(models.FriendLink).TableName()).All(&friendLinks)
	if err != nil {
		c.Data["json"] = models.GetRevaErr(err, "获取友链失败")
		c.ServeJSON()
		return
	}
	c.Data["json"] = models.GetRevaOk(friendLinks)
	c.ServeJSON()
}
